package com.example.smartlearning.aop;


import com.example.smartlearning.config.Annontation;
import com.example.smartlearning.entity.ResultEntity;
import net.jodah.expiringmap.ExpirationPolicy;
import net.jodah.expiringmap.ExpiringMap;
import jakarta.servlet.http.HttpServletRequest;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;

@Aspect
@Component
public class LimitRequestAspect {
    //在高并发的情况下，使用HashMap进行大量变更操作容易出现问题，但是ConcurrentHashMap是线程安全的。
    /**
     * 传统的HashMap 不管是用HashTable 或者是Collections.synchronizedMap() 方法都会给整个集合加锁 导致同一时间的其他操作阻塞
     * ConcurrentHashMap的优势在于兼顾性能和线程安全 JUC包下的一个集合类
     */
    private static ConcurrentHashMap<String, ExpiringMap<String, Integer>> book = new ConcurrentHashMap<>();

    // 定义切点
    // 让所有有@LimitRequest注解的方法都执行切面方法
    @Pointcut("@annotation(limitRequest)")//切入点 JoinPoint的集合 是程序中需要注入Advice的位置的集合 指明Advice要在什么样的条件下才能被触发
    public void excudeService(Annontation.LimitRequest limitRequest) {
    }

    @Around("excudeService(limitRequest)")//@Around注解可以用来在调用一个具体方法前和调用后来完成一些具体的任务。
    public Object doAround(ProceedingJoinPoint pjp, Annontation.LimitRequest limitRequest) throws Throwable {
        // 获得request对象
        RequestAttributes ra = RequestContextHolder.getRequestAttributes();
        //存储每个线程之间的私有资源，使之线程之间保证信息安全
        ServletRequestAttributes sra = (ServletRequestAttributes) ra;
        HttpServletRequest request = sra.getRequest();

        // 获取Map对象， 如果没有则返回默认值
        // 第一个参数是key， 第二个参数是默认值
        ExpiringMap<String, Integer> map = book.getOrDefault(request.getRequestURI(), ExpiringMap.builder().variableExpiration().build());
        Integer uCount = map.getOrDefault(request.getRemoteAddr(), 0);


        if (uCount >= limitRequest.count()) { // 超过次数，不执行目标方法
            //这里的返回对象类型根据controller方法的返回方式一致
            return ResultEntity.successlimit(0);
        } else if (uCount == 0) { // 第一次请求时，设置开始有效时间
            map.put(request.getRemoteAddr(), uCount + 1, ExpirationPolicy.CREATED, limitRequest.time(), TimeUnit.MILLISECONDS);
        } else { // 未超过次数， 记录数据加一
            map.put(request.getRemoteAddr(), uCount + 1);
        }
        book.put(request.getRequestURI(), map);

        // result的值就是被拦截方法的返回值
        Object result = pjp.proceed();

        return result;
    }
}
